// Copyright (c) Appeon. All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. 
// You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0.

using System.Data;
using System.Data.Common;
using JetBrains.Annotations;
using IBM.Data.DB2.Core;

namespace SnapObjects.Data.Informix 
{
    public class InformixTimeSpanTypeMapping : TimeSpanTypeMapping
    {
        private readonly DB2Type _db2Type;

        public InformixTimeSpanTypeMapping(
            [NotNull] string storeType, 
            DB2Type db2Type = DB2Type.Time,
            [CanBeNull] DbType? dbType = null)
            : base(storeType, dbType)
        {
            _db2Type = db2Type;
        }

        protected override void ConfigureParameter(DbParameter parameter, IAdoDbParameter adoDbParameter)
        {
            base.ConfigureParameter(parameter, adoDbParameter);

            // Workaround for a SQLClient bug
            if (parameter is DB2Parameter db2Parameter)
            {
                db2Parameter.DB2Type = _db2Type;
            }
        }
    }
}
